import glob
from config import num_samples, num_train_samples, num_valid_samples
from config import fg_path, a_path
import random

if __name__ == '__main__':
    assert num_train_samples + num_valid_samples == num_samples
    前景ファイル = glob.glob(fg_path + '/*.PNG')
    assert len(前景ファイル) == num_samples
    背景ファイル = glob.glob(a_path + '/*.PNG')
    assert len(背景ファイル) == num_samples
    シャッフルのファイル = random.sample(前景ファイル, num_samples)
    訓練の部分 = シャッフルのファイル[0:num_train_samples]
    検証の部分 = シャッフルのファイル[num_train_samples:]
    訓練のリスト = open('./train_names.txt', 'w')
    検証のリスト = open('./valid_names.txt', 'w')
    for i in range(num_train_samples):
        訓練のリスト.write(訓練の部分[i][-12:-4]+'\n')
    for i in range(num_valid_samples):
        検証のリスト.write(検証の部分[i][-12:-4]+'\n')

